import dataPretreatment as dataPretreatment

X, Y, wordNum, wordToID, words = dataPretreatment.pretreatment("dataset/QuanSongCi.txt")